package com.joker.controller.admin;

import com.joker.pojo.User;
import com.joker.service.UserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;

import javax.servlet.http.HttpSession;

/**
 * @Author JokerKS
 * @Date 2021/12/06 10:34
 * @Version 1.0
 */
@Controller
@RequestMapping("/admin")
public class LoginController {

	@Autowired
	UserService userService;

	@GetMapping
	public String loginPage() {
		return "admin/login";
	}

	@PostMapping("/login")
	public String login(@RequestParam String username, @RequestParam String password,
						HttpSession session, RedirectAttributes attributes) {
		User user = userService.checkUser(username, password);
		if (user != null) {
			user.setPassword(null);
			session.setAttribute("user", user);
			return "admin/index";
		} else {
			attributes.addFlashAttribute("message", "用户名或密码错误");
			return "redirect:/admin";
		}
	}

	@GetMapping("/loginOut")
	public String loginOut(HttpSession session) {
		session.removeAttribute("user");
		return "redirect:/admin";
	}

}
